import pandas as pd

# 读取 Excel 文件
df = pd.read_excel('./data/Cluster.xlsx')

# 打印列名以检查实际的列名
print(df.columns)

# 选择 x1 和 x2 列
df = df[['X1', 'X2']]

# 将 x1 和 x2 列的数据转换为嵌套列表
cluster_data = df.values.tolist()

# 生成 JavaScript 文件内容
js_content = "const clusterData = [\n"
for record in cluster_data:
    js_content += f"  [{record[0]}, {record[1]}],\n"
js_content = js_content.rstrip(",\n") + "\n];"

# 将内容写入文件
with open('clusterData.js', 'w', encoding='utf-8') as file:
    file.write(js_content)

print("JavaScript 文件已成功保存。")